package com.eobard.params;

import com.eobard.enums.PyInstallEnum;
import lombok.Builder;
import lombok.Data;

import java.util.ArrayList;
import java.util.List;

/**
 * @Author: Eobard Gu
 * @Date: 2025-09-30
 * @Description: Mnn模型转换器配置类
 */

@Data
@Builder
public class MnnConverterParams {
    //mnn文件的路径位置
    private String mnnModelPath;

    //mnn安装位置
    private String mnnInstallRoot;

    //mnn->json 和 json->mnn的中间临时保存位置
    @Builder.Default
    private String converterPath = System.getProperty("user.home") + "/mnn_converter_output";

    //指定修改md5的模型文件列表，默认修改全部
    private List<String> mnnModelNames;

    //指定为mnn->json的过程
    @Builder.Default
    private boolean forwardConverter = true;

    //安装依赖库的枚举变量
    @Builder.Default
    private PyInstallEnum installEnum=PyInstallEnum.PIP;

}
